#7. 保存模型并预测新数据
#7.1 保存模型
import joblib

# 保存模型和编码器
joblib.dump(best_model, '水果分类模型.pkl')
joblib.dump(encoder_color, '颜色编码器.pkl')

#7.2 加载模型并预测
# 加载模型和编码器
loaded_model = joblib.load('水果分类模型.pkl')
loaded_encoder = joblib.load('颜色编码器.pkl')

# 新数据示例
new_data = pd.DataFrame({
    '大小(cm)': [7.5, 20.0, 2.0],
    '颜色': ['红', '黄', '紫']
})

# 预处理新数据
new_data['颜色编码'] = loaded_encoder.transform(new_data['颜色'])
X_new = new_data[['大小(cm)', '颜色编码']]

# 预测
predictions = loaded_model.predict(X_new)
print("预测结果:", predictions)